# -*- coding:utf-8 -*-
# 1.数据存到文件里的格式
# 姓名，年龄，手机，身份证，学科
# 2.要确保手机身份证在文件里同样的数据只有一条，
# 可以在文件里用列表存放，用户输入相关列，就到列表里查询有没有重复的
# 3. 选学科时，给用户列出来选项，供选择

db_file="student_data.db"

def register_api():#注册信息用的函数
    stu_data={}#初始化一个字典，返回学员数据
    print("欢迎注册：".center(50,"-"))
    print("请完成注册")
    name=input("姓名：").strip()
    age=input("年龄：").strip()
    phone=input("手机").strip()
    id_num=input("身份证号：").strip()

    course_list=["Python开发","Linux云计算","网络安全","人工智能"]
    for index,coures in enumerate(course_list):
        print(f"{index}.{coures}")

    selected_course=input("请选择想学的课：")
    if selected_course.isdigit():#此处判断所输入内容是否是一个数字   函数作用为，是否只包含数字
        selected_course=int(selected_course)
        if selected_course>=0 and selected_course<len(course_list):#合法选项
            picked_coures=course_list[selected_course]#选中的课程
        else:
            exit("选项不合法。。。")
    else:
        exit("选项不合法。。。")
    stu_data["name"]=name
    stu_data["age"]=age
    stu_data["phone"]=phone
    stu_data["id_num"]=id_num
    stu_data["picked_coures"]=picked_coures
    return stu_data

def commit_to_db(filename,stu_data):
    '''
    把学院数据存到文件里
    :param filename:学员数据库文件
    :param stu_data:单个学员的数据
    :return:
    '''
    f=open(filename,"a")
    row=f"{stu_data['name']},{stu_data['age']},{stu_data['phone']},{stu_data['id_num']},{stu_data['picked_coures']}\n"
    #拼接成字符串
    f.write(row)#追加进文件
    f.close()#关闭文件

stu_data=register_api()
print(stu_data)
commit_to_db(db_file,stu_data)